/*
** Copyright (C) 2010 Calin Radoni
**
** This file is part of CHScanner.
**
** CHScanner is free software: you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation, either version 3 of the License, or
** (at your option) any later version.
**
** CHScanner is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with CHScanner. If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once
#include "scanmethod.h"

class ScanMethodDns : public ScanMethod
{
private:
	BYTE rBuffer1[18];
	BYTE rBuffer2[6];
	BYTE rBuffer3[256];
	BYTE rBuffer4[14];

protected:

	virtual int InternalInit(void);
	virtual int InternalExecute(void);

	virtual void ComputeRandomValues(void);
	bool SendDnsQueryRRPTR(MFullAddress*);
	bool SendDnsQueryRRPTR6(MFullAddress*);

	virtual void ProcessUdp(BYTE*, int);
	BYTE HexaCharToByte(char);

	/*
		returns:
			 1 = IPv6 address in query
			 0 = IPv4 address in query
			-1 = error
	*/
	int ExtractAddressFromResponse(BYTE*, int, DWORD*, in_addr6*, int*);

	BYTE dnsMAC[6];

	void FormatName(BYTE*, int);

public:
	ScanMethodDns(void);
	~ScanMethodDns(void);
};
